🐈 Cat Emotion Classification with EfficientNet

Import Modules

Set Hyperparameters

Set Device

Load Dataset (YOLOv7 Format)

Show Sample Images

Import EfficientNet-B5

Transfer Learning

EfficientNet-B5 Summary

Start Training

Set Loss and Optimizer

Train Loop

Plot Running Loss

Import Best Model

Evaluate with Validation Set

Visualize Trained Weights

Main Convolution Filters

Plot Feature Maps of the Main Convolution Layer

Evaluate Test Dataset

Load Test Dataset

Confusion Matrix

Plot Sample Prediction

Trace Pass Forwarding Procedure via Real-World Image

0️⃣ Load Image and Tensorize

1️⃣ Set-Up Hooks

2️⃣ Set Plotting Function and Save Step-by-Step Activation

3️⃣ Plot First Activation

4️⃣ Plot Third Activation

5️⃣ Plot Fifth Activation

6️⃣ Plot Seventh Activation

7️⃣ Plot Nineth Activation

8️⃣ Plot Classifier's Neuron Activation

📊 Show Final Prediction